/*
 * Copyright 2008-2012 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.broadleafcommerce.openadmin.server.service.artifact.upload;

import org.apache.commons.fileupload.FileUpload;
import org.springframework.web.multipart.MultipartException;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import javax.servlet.http.HttpServletRequest;

/**
 * Created by IntelliJ IDEA.
 * User: jfischer
 * Date: 9/2/11
 * Time: 12:07 PM
 * To change this template use File | Settings | File Templates.
 */
public class UploadMultipartResolver extends CommonsMultipartResolver {

    private static ThreadLocal<UploadProgressListener> progressListener = new ThreadLocal<UploadProgressListener>();

    @Override
    public void cleanupMultipart(MultipartHttpServletRequest request) {
        progressListener.get().setPercentDone(100D);
        super.cleanupMultipart(request);
    }

    @Override
    protected FileUpload prepareFileUpload(String encoding) {
        FileUpload response = super.prepareFileUpload(encoding);
        response.setProgressListener(progressListener.get());
        return response;
    }

    @Override
    public MultipartHttpServletRequest resolveMultipart(HttpServletRequest request) throws MultipartException {
        progressListener.set(new UploadProgressListener(request));
        return super.resolveMultipart(request);
    }

}
